import os
import shutil
import glob
import csv
import pinyin
chong=[]
with open('虫子统计.csv', 'r',encoding='utf-8') as f:
  reader = csv.reader(f)
  result = list(reader)

  for i in range(1,len(result)):
      # print(result[i])
      if int(result[i][1])>=59:
          chong.append(result[i][0])
print(chong)
print(len(chong))
def my_move(srcfn, dstdir):  ##定义移动函数，参数为待移动的文件路径和目标目录
    if not os.path.isfile(srcfn):  ##判断文件是否存在
        print('srcfn error')

    else:
        srcdir, fn = os.path.split(srcfn)  ##分离绝对路径和相对路径，获得文件名

        if not os.path.exists(dstdir):  ##如果目录不存在，创建目录
            os.makedirs(dstdir)

        dstfn = dstdir + fn  ##生成目录下的文件名
        shutil.move(srcfn, dstfn)  ##移动
#
#
filename = glob.glob('E:/T-test/zhijie/*.jpg')  ##获取当前目录下所有jpg格式的文件
# filename = os.listdir("E:/T-test/zhijie")
print(filename)
#
finish=[]
# src_fns = [fn for fn in fns if fn[-7:-4] == 'src']  ##获取当前目录下的所有原始数据，存为一个list
for i in filename:
        # print(pinyin.get(a, format='strip', delimiter=" ").replace(' ',''))
        for a in chong:
            if pinyin.get(a, format='strip', delimiter=" ").replace(' ','') in i:
                finish.append(i)
#
print(finish)
print(len(finish))
#
# for ind in range(len(finish)):  ##循环移动所有文件
#
#     my_move(finish[ind], 'E:/T-test/not-need-strength/')
